package com.note.feng.leetcode.algorithms.easy.eleven;

import java.util.HashMap;
import java.util.Map;

public class ThousandHundredTwentyEight {
    /**
     * 1128 等价多米诺骨牌对的数量
     * 给你一个由一些多米诺骨牌组成的列表 dominoes。
     *
     * 如果其中某一张多米诺骨牌可以通过旋转 0 度或 180 度得到另一张多米诺骨牌，我们就认为这两张牌是等价的。
     *
     * 形式上，dominoes[i] = [a, b] 和 dominoes[j] = [c, d] 等价的前提是 a==c 且 b==d，或是 a==d 且 b==c。
     *
     * 在 0 <= i < j < dominoes.length 的前提下，找出满足 dominoes[i] 和 dominoes[j] 等价的骨牌对 (i, j) 的数量。
     *
     * 示例：
     *
     * 输入：dominoes = [[1,2],[2,1],[3,4],[5,6]]
     * 输出：1
     *
     * 提示：
     *
     * 1 <= dominoes.length <= 40000
     * 1 <= dominoes[i][j] <= 9
     *
     * 来源：力扣（LeetCode）
     * 链接：https://leetcode.cn/problems/number-of-equivalent-domino-pairs
     */
    /**
     * 解法：计数+求和
     * 1.计数：
     *   遍历二维数组，每个一维数组都可以组成一个两位数（11-99），按照十位大个位小的原则，组成两位数，
     *   存入到 map 中，两位数为 key，value 为 key 出现的次数；
     * 2.求和
     *   遍历 map，若 key 出现了2次及以上，累加数量；
     *   2次：1
     *   3次：3
     *   4次：6
     *   根据出现的次数不同，可以推算出符合等差数列求和公式：s(n) = n + n * (n - 1) / 2; n为项数
     *   对于数量为 n 的骨牌对于的项数为 n - 1
     * @param dominoes
     * @return
     */
    public int numEquivDominoPairs(int[][] dominoes) {
        int res = 0;
        Map<Integer, Integer> map = new HashMap<>();
        for(int [] d : dominoes){
            int c = d[0] > d[1] ? d[0] * 10 + d[1] : d[1] * 10 + d[0];
            map.put(c, map.getOrDefault(c, 0) + 1);
        }
        for(int k : map.keySet()){
            int v = map.get(k);
            if(v > 1){
                res += (v - 1) + (v - 1) * (v - 2) / 2;
            }
        }
        return res;
    }
}
